package com.atguigu.daijia.order.service.impl;

import com.alibaba.cloud.commons.lang.StringUtils;
import com.atguigu.daijia.order.service.TestService;
import jakarta.annotation.Resource;
import org.redisson.api.RLock;
import org.redisson.api.RedissonClient;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Service;

import java.util.Collections;
import java.util.UUID;
import java.util.concurrent.TimeUnit;

/**
 * @Author XiangLiang
 * @Since 2024/12/5 14:47
 * @Description: TestServiceImpl类
 */
@Service
public class TestServiceImpl implements TestService {
    @Resource
    private StringRedisTemplate stringRedisTemplate;

    @Resource
    private RedissonClient redissonClient;

    @Override
    public void testLock() {
        // 本地锁测试
        this.locationLock();

        // 采用SpringDataRedis实现分布式锁
        this.redisLock();

        // 采用SpringDataRedis实现分布式锁，添加Lua脚本保证原子性删除
        this.redisLockAtomicDelete();

        // 采用Redisson实现分布式锁
        this.redissonLock();
    }

    /**
     * 本地锁测试
     */
    public void locationLock() {
        // 查询Redis中的num值
        String value = this.stringRedisTemplate.opsForValue().get("num");
        // 没有该值return
        if (StringUtils.isBlank(value)) {
            return;
        }
        // 有值就转成成int
        int num = Integer.parseInt(value);
        // 把Redis中的num值+1
        this.stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));
    }

    /**
     * 采用SpringDataRedis实现分布式锁
     * 原理：执行业务方法前先尝试获取锁（setnx存入key val），如果获取锁成功再执行业务代码，业务执行完毕后将锁释放(del key)
     * 此版：有过期时间（防死锁）和UUID防误删其他的锁；问题：删除缺乏原子性
     */
    public void redisLock() {
        // 0.先尝试获取锁 setnx key val
        String uuid = UUID.randomUUID().toString();
        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent("lock", uuid, 3, TimeUnit.SECONDS);
        if (Boolean.TRUE.equals(flag)) {
            // 获取锁成功，执行业务代码
            // 1.先从redis中通过key num获取值  key提前手动设置 num 初始值：0
            String value = stringRedisTemplate.opsForValue().get("num");
            // 2.如果值为空则非法直接返回即可
            if (StringUtils.isBlank(value)) {
                return;
            }
            // 3.对num值进行自增加一
            int num = Integer.parseInt(value);
            stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));

            // 4.将锁释放
            if (uuid.equals(stringRedisTemplate.opsForValue().get("lock"))) {
                stringRedisTemplate.delete("lock");
            }
        } else {
            try {
                Thread.sleep(100);
                this.redisLock();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 采用SpringDataRedis实现分布式锁，添加Lua脚本保证原子性删除
     * 问题：无自动续期
     */
    public void redisLockAtomicDelete() {
        // 0.先尝试获取锁 setnx key val
        String uuid = UUID.randomUUID().toString();
        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent("lock", uuid, 3, TimeUnit.SECONDS);
        if (Boolean.TRUE.equals(flag)) {
            // 获取锁成功，执行业务代码
            // 1.先从redis中通过key num获取值  key提前手动设置 num 初始值：0
            String value = stringRedisTemplate.opsForValue().get("num");
            // 2.如果值为空则非法直接返回即可
            if (StringUtils.isBlank(value)) {
                return;
            }
            // 3.对num值进行自增加一
            int num = Integer.parseInt(value);
            stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));

            // 4.1 先创建脚本对象 DefaultRedisScript泛型脚本语言返回值类型 Long 0：失败 1：成功
            DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();
            // 4.2设置脚本文本
            String script = "if redis.call(\"get\",KEYS[1]) == ARGV[1]\n" +
                    "then\n" +
                    "    return redis.call(\"del\",KEYS[1])\n" +
                    "else\n" +
                    "    return 0\n" +
                    "end";
            redisScript.setScriptText(script);
            // 4.3 设置响应类型
            redisScript.setResultType(Long.class);
            // 执行脚本参数 参数1：脚本对象封装lua脚本，参数二：lua脚本中需要key参数（KEYS[i]）  参数三：lua脚本中需要参数值 ARGV[i]
            stringRedisTemplate.execute(redisScript, Collections.singletonList("lock"), uuid);
        } else {
            try {
                Thread.sleep(100);
                this.redisLock();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 使用Redisson实现分布式锁
     * 开发步骤：
     * 1.使用RedissonClient客户端对象 创建锁对象
     * 2.调用获取锁方法
     * 3.执行业务逻辑
     * 4.将锁释放
     */
    public void redissonLock() {
        // 0.创建锁对象
        RLock lock = redissonClient.getLock("lock");

        // 0.1 尝试加锁
        // 0.1.1 lock() 阻塞等待一直到获取锁,默认锁有效期30s
        lock.lock();

        // 1.先从redis中通过key num获取值  key提前手动设置 num 初始值：0
        String value = stringRedisTemplate.opsForValue().get("num");
        // 2.如果值为空则非法直接返回即可
        if (StringUtils.isBlank(value)) {
            return;
        }
        // 3.对num值进行自增加一
        int num = Integer.parseInt(value);
        stringRedisTemplate.opsForValue().set("num", String.valueOf(++num));

        // 4.将锁释放
        lock.unlock();
    }
}
